#!/bin/bash

# 1. 检测包管理器并安装 zsh
if command -v apt &> /dev/null; then
  echo "1. 检测到 APT 包管理器（Ubuntu/Debian），正在安装 zsh..."
  sudo apt update
  sudo apt install -y zsh
elif command -v yum &> /dev/null; then
  echo "1. 检测到 YUM 包管理器（CentOS/RHEL），正在安装 zsh..."
  sudo yum install -y zsh
else
  echo "1. 未检测到 apt 或 yum 包管理器，请手动安装 zsh。"
  exit 1
fi

# 2. 安装 git（oh-my-zsh 依赖 git）
if command -v apt &> /dev/null; then
  sudo apt install -y git
elif command -v yum &> /dev/null; then
  sudo yum install -y git
fi

# 3. 安装 oh-my-zsh（从 Gitee 安装）
echo "3. 正在安装 Oh My Zsh（从 Gitee 官方镜像）..."
sh -c "$(curl -fsSL https://gitee.com/pocmon/ohmyzsh/raw/master/tools/install.sh)"

# 4. （可选）提示用户可以手动运行 zsh，但不自动执行
echo "4. （可选）Zsh 已安装，你可以手动运行 'zsh' 命令进入 zsh 环境体验。"
echo "   如果你想现在体验，可以退出脚本后手动运行：zsh"

# 5. 设置用户默认 Shell 为 zsh（需用户密码，永久生效）
echo "5. 正在设置默认 Shell 为 zsh（需要输入当前用户密码）..."
if command -v chsh &> /dev/null; then
  chsh -s $(which zsh)
  echo "5. 默认 Shell 已设置为 zsh！（可能需要重新登录或重启终端生效）"
else
  echo "5. 未找到 chsh 命令，无法设置默认 Shell。请手动设置。"
fi

echo "6. Zsh + Oh My Zsh 安装与配置完成！"
echo "6. 提示：你可能需要重新登录终端或重启系统，才能让默认 Shell 生效。"